k-nearest neighbor algorithm example python